#include "mystdio.h"
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <assert.h>
#include <string.h>

#define MODE 0666

_FILE* _fopen(const char* filename, const char* flag)
{
    assert(filename);
    assert(flag);
    
    int f = 0; //标志位
    int fd = 0; //接收open返回值
    umask(0);
    
    if(strcmp(flag, "w") == 0)
    {
        f = (O_CREAT|O_WRONLY|O_TRUNC);
        fd = open(filename, f, MODE);
    }
    else if(strcmp(flag, "a") == 0)
    {
        f = (O_CREAT|O_WRONLY|O_APPEND);
        fd = open(filename, f, MODE);
    }
    else if(strcmp(flag, "r") == 0)
    {
        f = O_WRONLY;
        fd = open(filename, f, MODE);
    }
    else
    {
        printf("输入错误\n");
        return NULL;
    }

    _FILE* fp = (_FILE*)malloc(sizeof(_FILE));
    fp->fileno = fd;
    //fp->flag = FLASH_LINE;
    fp->flag = FLASH_ALL;
    fp->out_pos = 0;
    return fp;
}

int _fwrite(_FILE* fp, const char* s, size_t len)
{
    assert(fp);
    assert(s);

    memcpy(&fp->out_buffer[fp->out_pos], s, fp->out_pos);
    fp->out_pos += len; //此处没有处理超出缓冲区的情况
    
    if(fp->flag & FLASH_LINE)
    {
        if(fp->out_buffer[fp->out_pos - 1] =='\n') //这里没有考虑一行代码出现多个\n的问题
        {
            write(fp->fileno, fp->out_buffer, fp->out_pos);
            fp->out_pos = 0;
        }
    }
    else if(fp->flag &  FLASH_NOW) //需要处理写入失败的情况
    {
        write(fp->fileno, fp->out_buffer, fp->out_pos);
        fp->out_pos = 0;
    }
    else if(fp->flag & FLASH_ALL)
    {
        if(fp->out_pos == SIZE)
        {
            write(fp->fileno, fp->out_buffer, fp->out_pos);
            fp->out_pos = 0;
        }
    }
    //write(fp->fileno, s, len);

    return len;
}

void _fclose(_FILE* fp)
{
    assert(fp);
    if(fp->out_pos > 0)
    {
        write(fp->fileno, fp->out_buffer, fp->out_pos); //关闭文件前强制写入
        fp->out_pos = 0;
    }
    close(fp->fileno);
    free(fp);
}
